import os

from torchvision import datasets

# 带用户名/密码的代理（如果不需要认证，去掉 username:password@ 部分）
os.environ['HTTP_PROXY'] = 'http://localhost:7890'
os.environ['HTTPS_PROXY'] = 'https://localhost:7890'
# 下载CIFAR-10
train_dataset = datasets.CIFAR10(root='./data', train=True, download=True)
# 下载CIFAR-100
test_dataset = datasets.CIFAR100(root='./data', train=False, download=True)